set INPUT_DATATYPE {ac::bfloat16}
set WEIGHT_DATATYPE {ac::bfloat16}
set OUTPUT_DATATYPE {ac::bfloat16}

set DIMENSION {4}

set INPUT_BUFFER_SIZE 1024
set WEIGHT_BUFFER_SIZE 1024
set ACCUM_BUFFER_SIZE 1024
